package org.sicau.votesys.util;

import org.apache.poi.hssf.usermodel.*;

/**
 * @Author beifengtz
 * @Site www.beifengtz.com
 * @Date Created in 16:18 2018/11/5
 * @Description: Excel表格工具类
 */
public class ExcelUtil {

    public static HSSFWorkbook getHSSFWorkbook(String sheetName,String[] title,String [][] values,HSSFWorkbook wb){
        // 1、创建一个HSSFWorkbook,对应一个Excel文件
        if(wb == null){
            wb = new HSSFWorkbook();
        }
        // 2、在workbook里面添加一个sheet
        HSSFSheet sheet = wb.createSheet(sheetName);

        // 3、在sheet中添加表头第0行
        HSSFRow row = sheet.createRow(0);

        // 4、创建单元格，并设置值表头，设置表头居中
        HSSFCellStyle style = wb.createCellStyle();
        style.setAlignment(HSSFCellStyle.ALIGN_CENTER);// 居中表格

        // 声明列对象
        HSSFCell cell = null;

        // 创建标题
        for (int i=0;i<title.length;i++){
            cell = row.createCell(i);
            cell.setCellValue(title[i]);
            cell.setCellStyle(style);
        }

        // 创建内容
        for(int i=0;i<values.length;i++){
            row = sheet.createRow(i+1);
            for (int j=0;j<values[i].length;j++){
                // 将内容按顺序付给对应的列对象
                row.createCell(j).setCellValue(values[i][j]);
            }
        }
        return wb;
    }
}
